package com.fengye.i18n.config;

import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler;
import com.fengye.i18n.hodler.TenantContextHolder;
import lombok.AccessLevel;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import net.sf.jsqlparser.expression.Expression;
import net.sf.jsqlparser.expression.NullValue;
import net.sf.jsqlparser.expression.StringValue;
import org.springframework.beans.factory.annotation.Autowired;

import java.util.Objects;

/**
 * 租户维护处理器
 */
@Slf4j
@RequiredArgsConstructor(access = AccessLevel.PUBLIC)
public class MyTenantHandler implements TenantLineHandler {
    
    private final MyTenantConfigProperties properties;
    
    /**
     * 获取租户 ID 值表达式，只支持单个 ID 值
     *
     * @return 租户 ID 值表达式
     */
    @Override
    public Expression getTenantId() {
        String tenantId = TenantContextHolder.getTenantId();
        log.debug("当前租户为 >> {}", tenantId);
        if (StrUtil.isBlank(tenantId)) {
            return new NullValue(); // 数据库的值
        }
        return new StringValue(tenantId);
    }
    
    /**
     * 获取租户字段名
     *
     * @return 租户字段名
     */
    @Override
    public String getTenantIdColumn() {
        return properties.getColumn();
    }
    
    /**
     * 根据表名判断是否忽略拼接多租户条件
     * <p>
     * 默认都要进行解析并拼接多租户条件
     *
     * @param tableName 表名
     * @return 是否忽略, true:表示忽略，false:需要解析并拼接多租户条件
     */
    @Override
    public boolean ignoreTable(String tableName) {
        if (Objects.nonNull(MybatisTenantContext.get())) {
            log.info("是否做租户隔离:{}", MybatisTenantContext.get());
            if (MybatisTenantContext.get()) {
                return Boolean.TRUE;
            }
            return ignoreByTableName(tableName);
        }
        return ignoreByTableName(tableName);
    }
    
    private boolean ignoreByTableName(String tableName) {
        String tenantId = TenantContextHolder.getTenantId();
        // 租户中ID 为空，查询全部，不进行过滤
        if (StrUtil.isBlank(tenantId)) {
            System.out.println("tenant 为空,应该忽略才对");
            return Boolean.TRUE;
        }
        return !properties.getTables().contains(tableName);
    }
}